#用法https://zhuanlan.zhihu.com/p/26821232
import numpy as np
import random
#第一个参数5,表示从0-4选择1个数，第二个参数表示要生成几个数，第三个p表示每个数被生成的概率
print(np.random.choice(5, 2, p=[0.1, 0.3, 0.2, 0.3, 0.1]));